from hey.agents.basic.core import BasicAgent, Agent_single_thread

registered_agents = {
    'basic': BasicAgent,
    'single_thread': Agent_single_thread,  # Harmony OS HiShell doesn't support threading for now
}


def get_agent(config, environment):
    return registered_agents[config.type](config, environment)
